package com.zzy.edu.listenr;

import com.alibaba.excel.context.AnalysisContext;
import com.alibaba.excel.event.AnalysisEventListener;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.zzy.edu.entity.EduSubject;
import com.zzy.edu.entity.vo.excel.SubjectData;
import com.zzy.edu.service.EduSubjectService;
import com.zzy.servicebase.exceptionhandler.GuliException;

/**
 * @Author KAZU
 * @Date 2023/1/11 19:00
 * @Version 1.0
 */
@SuppressWarnings({"all"})
public class SubjectExcelLister extends AnalysisEventListener<SubjectData> {
	
	// 因为SubjectExcelLister不能交给spring管理，需要手动new，不能注入其他对象，不能实现数据库操作
	
	public EduSubjectService subjectService;
	
	public SubjectExcelLister() {}
	
	public SubjectExcelLister(EduSubjectService subjectService) {
		this.subjectService = subjectService;
	}
	
	// 读取excel内容
	@Override
	public void invoke(SubjectData subjectData, AnalysisContext analysisContext) {
		if(subjectData == null) {
			throw new GuliException(20001, "文件数据为空");
		}
		// 一行一行读取，每次读取有两个值，第一个值一级分类，第二个值二级分类
		EduSubject existOneSubject = this.existOneSubject(subjectService, subjectData.getOneSubjectName());
		if(existOneSubject == null) {	// 没有相同的一级分类，进行添加
			existOneSubject = new EduSubject();
			existOneSubject.setParentId("0");
			existOneSubject.setTitle(subjectData.getOneSubjectName());
			subjectService.save(existOneSubject);
		}
		
		String pid = existOneSubject.getId();
		EduSubject existTwoSubject = this.existTwoSubject(subjectService, subjectData.getTwoSubjectName(), pid);
		if(existTwoSubject == null) {	// 没有相同的二级分类，进行添加
			existTwoSubject = new EduSubject();
			existTwoSubject.setParentId(pid);
			existTwoSubject.setTitle(subjectData.getTwoSubjectName());
			subjectService.save(existTwoSubject);
		}
	}
	
	/**
	 * 判断一级分类不能重复添加
	 * */
	public EduSubject existOneSubject(EduSubjectService subjectService, String name) {
		QueryWrapper<EduSubject> wrapper = new QueryWrapper<>();
		wrapper.eq("title", name);
		wrapper.eq("parent_id", "0");
		EduSubject oneSubject = subjectService.getOne(wrapper);
		return oneSubject;
	}
	
	/**
	 * 判断一级分类不能重复添加
	 * */
	public EduSubject existTwoSubject(EduSubjectService subjectService, String name, String pid) {
		QueryWrapper<EduSubject> wrapper = new QueryWrapper<>();
		wrapper.eq("title", name);
		wrapper.eq("parent_id", pid);
		EduSubject twoSubject = subjectService.getOne(wrapper);
		return twoSubject;
	}
	
	@Override
	public void doAfterAllAnalysed(AnalysisContext analysisContext) {
	
	}
}
